package controller;

import entidade.TreinoPreTreino;
import java.util.List;
import javax.faces.model.ListDataModel;
import org.primefaces.model.SelectableDataModel;

/**
 *
 * @author thais
 */
public class PreTreinoDataModel extends ListDataModel<TreinoPreTreino> implements SelectableDataModel<TreinoPreTreino> {

    public PreTreinoDataModel(List<TreinoPreTreino> data) {
        super(data);
    }

    @Override
    public Object getRowKey(TreinoPreTreino t) {
        return t.getPreTreino().getIdPreTreino();
    }

    @Override
    public TreinoPreTreino getRowData(String string) {
        List<TreinoPreTreino> listPreTreino = (List<TreinoPreTreino>) getWrappedData();

        for (TreinoPreTreino preTreino : listPreTreino) {
            if (preTreino.getPreTreino().getIdPreTreino().equals(Long.parseLong(string))) {
                return preTreino;
            }
        }
        return null;
    }
}
